package com.springboot.util;

import org.apache.commons.codec.digest.DigestUtils;

import java.io.UnsupportedEncodingException;
import java.security.SignatureException;
import java.util.Collections;
import java.util.List;

/**
 * @author zhenwei.wang 2019/7/5
 */
public class MD5Utils {

    public static String sign(byte[] bytes) {
        return DigestUtils.md5Hex(bytes);
    }

    /**
     * MD5 签名
     * @param text
     * @param charset 指定编码格式
     * @return
     */
    public static String sign(String text, String charset) {
        return DigestUtils.md5Hex(getContentBytes(text, charset));
    }

    /**
     * md5签名
     * 编码 UTF-8
     * @param text
     * @return
     */
    public static String sign(String text) {
        return DigestUtils.md5Hex(getContentBytes(text, "UTF-8"));
    }

    /**
     * 签名字符串
     * @param text 需要签名的字符串
     * @param key 密钥
     * @param input_charset 编码格式
     * @return 签名结果
     */
    public static String sign(String text, String key, String input_charset) {
        text = text + key;
        return DigestUtils.md5Hex(getContentBytes(text, input_charset));
    }

    /**
     * 签名字符串
     * @param text 需要签名的字符串
     * @param sign 签名结果
     * @param key 密钥
     * @param charset 编码格式
     * @return 签名结果
     */
    public static boolean verify(String text, String sign, String key, String charset) {
        text = text + key;
        String sign_ = DigestUtils.md5Hex(getContentBytes(text, charset));
        if(sign_.equals(sign)) {
            return true;
        }
        else {
            return false;
        }
    }

    /**
     * @param content
     * @param charset
     * @return
     * @throws SignatureException
     * @throws UnsupportedEncodingException
     */
    private static byte[] getContentBytes(String content, String charset) {
        if (charset == null || "".equals(charset)) {
            return content.getBytes();
        }
        try {
            return content.getBytes(charset);
        } catch (UnsupportedEncodingException e) {
            throw new RuntimeException("MD5签名过程中出现错误,指定的编码集不对,您目前指定的编码集是:" + charset);
        }
    }

    /**
     * 对一组字符串获取签名
     * @param text
     * @return
     */
    public static String sign(List<String> text) {
        if (text == null && text.size() == 0) {
            return "";
        }
        Collections.sort(text);
        StringBuilder builder = new StringBuilder();
        for (String s : text) {
            builder.append(s);
        }
        return sign(builder.toString());
    }

}
